package com.ruoyi.file.controller;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import com.ruoyi.common.core.domain.R;
import com.ruoyi.file.utils.AliyunOssUtils;

/**
 * OSS文件上传控制器
 * 
 * @author ruoyi
 */
@RestController
@RequestMapping("/oss")
public class OssController
{
    private static final Logger log = LoggerFactory.getLogger(OssController.class);

    @Autowired
    private AliyunOssUtils aliyunOssUtils;

    /**
     * 上传单个文件到OSS
     */
    @PostMapping("/upload")
    public R<Object> uploadFile(MultipartFile file)
    {
        try
        {
            if (file == null || file.isEmpty())
            {
                return R.fail("上传文件不能为空");
            }

            // 检查文件大小，不能超过10MB
            long maxSize = 10 * 1024 * 1024L; // 10MB
            if (file.getSize() > maxSize)
            {
                return R.fail("上传文件大小不能超过10MB");
            }

            // 上传文件到OSS
            String fileUrl = aliyunOssUtils.uploadFile(file);
            
            Map<String, Object> resultData = new HashMap<>();
            resultData.put("url", fileUrl);
            resultData.put("fileName", file.getOriginalFilename());
            resultData.put("fileSize", file.getSize());
            resultData.put("message", "文件上传成功");
            
            return R.ok(resultData);
        }
        catch (Exception e)
        {
            log.error("文件上传失败", e);
            return R.fail("文件上传失败：" + e.getMessage());
        }
    }

    /**
     * 批量上传文件到OSS（优化版）
     */
    @PostMapping("/uploads")
    public R<Object> uploadFiles(@RequestParam("files") List<MultipartFile> files)
    {
        try
        {
            if (files == null || files.isEmpty())
            {
                return R.fail("上传文件不能为空");
            }

            // 限制批量上传文件数量
            if (files.size() > 20)
            {
                return R.fail("批量上传文件数量不能超过20个");
            }

            List<Map<String, Object>> successList = new ArrayList<>();
            List<Map<String, Object>> failList = new ArrayList<>();
            long maxSize = 10 * 1024 * 1024L; // 10MB
            long totalSize = 0;

            // 预检查所有文件
            for (MultipartFile file : files)
            {
                if (file != null && !file.isEmpty())
                {
                    totalSize += file.getSize();
                    if (file.getSize() > maxSize)
                    {
                        Map<String, Object> failInfo = new HashMap<>();
                        failInfo.put("fileName", file.getOriginalFilename());
                        failInfo.put("reason", "文件大小超过10MB");
                        failInfo.put("fileSize", file.getSize());
                        failList.add(failInfo);
                    }
                }
            }

            // 检查总文件大小（不超过100MB）
            if (totalSize > 100 * 1024 * 1024L)
            {
                return R.fail("批量上传文件总大小不能超过100MB");
            }

            // 批量上传文件
            List<String> uploadedFiles = new ArrayList<>(); // 用于回滚
            
            for (MultipartFile file : files)
            {
                if (file != null && !file.isEmpty() && file.getSize() <= maxSize)
                {
                    try
                    {
                        // 上传文件到OSS
                        String fileUrl = aliyunOssUtils.uploadFile(file);
                        
                        Map<String, Object> successInfo = new HashMap<>();
                        successInfo.put("url", fileUrl);
                        successInfo.put("fileName", file.getOriginalFilename());
                        successInfo.put("fileSize", file.getSize());
                        successInfo.put("uploadTime", System.currentTimeMillis());
                        successList.add(successInfo);
                        
                        uploadedFiles.add(aliyunOssUtils.extractFileNameFromUrl(fileUrl));
                        
                        log.info("文件上传成功: {}", file.getOriginalFilename());
                    }
                    catch (Exception e)
                    {
                        Map<String, Object> failInfo = new HashMap<>();
                        failInfo.put("fileName", file.getOriginalFilename());
                        failInfo.put("reason", "上传失败: " + e.getMessage());
                        failInfo.put("fileSize", file.getSize());
                        failList.add(failInfo);
                        
                        log.error("文件上传失败: {}, 错误: {}", file.getOriginalFilename(), e.getMessage());
                    }
                }
            }

            // 构建返回结果
            Map<String, Object> resultData = new HashMap<>();
            resultData.put("successCount", successList.size());
            resultData.put("failCount", failList.size());
            resultData.put("totalCount", files.size());
            resultData.put("successList", successList);
            resultData.put("failList", failList);
            resultData.put("totalSize", totalSize);
            
            if (successList.size() > 0)
            {
                resultData.put("message", String.format("批量上传完成：成功%d个，失败%d个", 
                    successList.size(), failList.size()));
                return R.ok(resultData);
            }
            else
            {
                resultData.put("message", "所有文件上传失败");
                return R.fail("所有文件上传失败");
            }
        }
        catch (Exception e)
        {
            log.error("批量文件上传异常", e);
            return R.fail("批量文件上传异常：" + e.getMessage());
        }
    }


    /**
     * 删除OSS文件
     */
    @DeleteMapping("/delete")
    public R<Object> deleteFile(@RequestParam String fileUrl)
    {
        try
        {
            if (fileUrl == null || fileUrl.trim().isEmpty())
            {
                return R.fail("文件URL不能为空");
            }

            // 从URL中提取文件名
            String fileName = aliyunOssUtils.extractFileNameFromUrl(fileUrl);
            if (fileName == null)
            {
                return R.fail("无效的文件URL");
            }

            // 删除文件
            boolean success = aliyunOssUtils.deleteFile(fileName);
            
            if (success)
            {
                return R.ok("文件删除成功");
            }
            else
            {
                return R.fail("文件删除失败");
            }
        }
        catch (Exception e)
        {
            log.error("文件删除失败", e);
            return R.fail("文件删除失败：" + e.getMessage());
        }
    }

    /**
     * 检查文件是否存在
     */
    @GetMapping("/exists")
    public R<Object> checkFileExists(@RequestParam String fileUrl)
    {
        try
        {
            if (fileUrl == null || fileUrl.trim().isEmpty())
            {
                return R.fail("文件URL不能为空");
            }

            // 从URL中提取文件名
            String fileName = aliyunOssUtils.extractFileNameFromUrl(fileUrl);
            if (fileName == null)
            {
                return R.fail("无效的文件URL");
            }

            // 检查文件是否存在
            boolean exists = aliyunOssUtils.doesObjectExist(fileName);
            
            Map<String, Object> resultData = new HashMap<>();
            resultData.put("exists", exists);
            resultData.put("fileName", fileName);
            
            return R.ok(resultData);
        }
        catch (Exception e)
        {
            log.error("检查文件失败", e);
            return R.fail("检查文件失败：" + e.getMessage());
        }
    }

    /**
     * 获取文件访问URL
     */
    @GetMapping("/url")
    public R<Object> getFileUrl(@RequestParam String fileName)
    {
        try
        {
            if (fileName == null || fileName.trim().isEmpty())
            {
                return R.fail("文件名不能为空");
            }

            // 获取文件访问URL
            String fileUrl = aliyunOssUtils.getFileUrl(fileName);
            
            if (fileUrl != null)
            {
                Map<String, Object> resultData = new HashMap<>();
                resultData.put("url", fileUrl);
                resultData.put("fileName", fileName);
                
                return R.ok(resultData);
            }
            else
            {
                return R.fail("获取文件URL失败");
            }
        }
        catch (Exception e)
        {
            log.error("获取文件URL失败", e);
            return R.fail("获取文件URL失败：" + e.getMessage());
        }
    }


    /**
     * 批量上传文件并返回拼接路径（专用于数据库存储）
     */
    @PostMapping("/uploadsP")
    public R<Object> uploadFilesForDatabase(@RequestParam("files") List<MultipartFile> files,
                                          @RequestParam(value = "separator", defaultValue = "|") String separator)
    {
        try
        {
            if (files == null || files.isEmpty())
            {
                return R.fail("上传文件不能为空");
            }

            // 限制批量上传文件数量
            if (files.size() > 20)
            {
                return R.fail("批量上传文件数量不能超过20个");
            }

            List<String> successUrls = new ArrayList<>();
            List<String> failedFiles = new ArrayList<>();
            long maxSize = 10 * 1024 * 1024L; // 10MB
            long totalSize = 0;

            // 预检查所有文件
            for (MultipartFile file : files)
            {
                if (file != null && !file.isEmpty())
                {
                    totalSize += file.getSize();
                    if (file.getSize() > maxSize)
                    {
                        failedFiles.add(file.getOriginalFilename() + "(文件过大)");
                    }
                }
            }

            // 检查总文件大小（不超过100MB）
            if (totalSize > 100 * 1024 * 1024L)
            {
                return R.fail("批量上传文件总大小不能超过100MB");
            }

            // 批量上传文件
            for (MultipartFile file : files)
            {
                if (file != null && !file.isEmpty() && file.getSize() <= maxSize)
                {
                    try
                    {
                        // 上传文件到OSS
                        String fileUrl = aliyunOssUtils.uploadFile(file);
                        successUrls.add(fileUrl);
                        
                        log.info("数据库存储上传成功: {}", file.getOriginalFilename());
                    }
                    catch (Exception e)
                    {
                        failedFiles.add(file.getOriginalFilename() + "(上传失败)");
                        log.error("数据库存储上传失败: {}, 错误: {}", file.getOriginalFilename(), e.getMessage());
                    }
                }
            }

            // 构建返回结果
            Map<String, Object> resultData = new HashMap<>();
            
            if (!successUrls.isEmpty())
            {
                // 用指定分隔符拼接所有成功上传的图片路径
                String concatenatedUrls = String.join(separator, successUrls);
                
                resultData.put("urls", concatenatedUrls);           // 拼接后的字符串（用于数据库存储）
                resultData.put("urlList", successUrls);             // 原始URL列表
                resultData.put("separator", separator);             // 使用的分隔符
                resultData.put("successCount", successUrls.size()); // 成功上传数量
                resultData.put("failCount", failedFiles.size());    // 失败数量
                resultData.put("totalCount", files.size());         // 总文件数
                resultData.put("totalSize", totalSize);             // 总文件大小
                
                if (!failedFiles.isEmpty())
                {
                    resultData.put("failedFiles", failedFiles);
                    resultData.put("message", String.format("批量上传完成：成功%d个，失败%d个。拼接路径已生成", 
                        successUrls.size(), failedFiles.size()));
                }
                else
                {
                    resultData.put("message", String.format("批量上传全部成功：共%d个文件。拼接路径已生成", 
                        successUrls.size()));
                }
                
                return R.ok(resultData);
            }
            else
            {
                resultData.put("urls", "");                         // 空字符串
                resultData.put("successCount", 0);
                resultData.put("failCount", failedFiles.size());
                resultData.put("totalCount", files.size());
                resultData.put("failedFiles", failedFiles);
                resultData.put("message", "所有文件上传失败，无法生成拼接路径");
                
                return R.fail("所有文件上传失败");
            }
        }
        catch (Exception e)
        {
            log.error("数据库存储批量上传异常", e);
            return R.fail("批量上传异常：" + e.getMessage());
        }
    }

}